package com.briup.oagg.facade.web.vm.model;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.util.List;

@ApiModel(description = "评估结果模型")
@Data
public class ClassificationResultVM {
    @ApiModelProperty(value = "训练id")
    private String taskId;

    @ApiModelProperty(value = "训练状态")
    private int trainStatus;

    @ApiModelProperty(value = "精确率")
    private Double accuracy;

    @ApiModelProperty(value = "准确率")
    private List<Double> precision;

    @ApiModelProperty(value = "召回率")
    private List<Double> recall;

    @ApiModelProperty(value = "f1_score")
    private List<Double> f1Score;
    @ApiModelProperty(value = "gScore")
    private List<Double> gScore;
}